/*
 *
 * Copyright (C) 2012 Walter Lütgenau
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * 
 */
package offset.nodes.client.tree.view;

import java.awt.Component;
import javax.swing.Icon;
import javax.swing.JTree;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeCellRenderer;
import offset.nodes.client.tree.model.DetailPanelTreeNodeType;
import offset.nodes.client.model.TreeNodeUserObject;
import offset.nodes.client.tree.model.TreeNodeType;

public class TreeNodeTypeRenderer extends DefaultTreeCellRenderer {
    TreeNodeType[] types;
    
    public TreeNodeTypeRenderer(TreeNodeType[] types) {
        this.types = types;
    }
    
    public Icon getIcon(Object userObject) {
        if ( !(userObject instanceof TreeNodeUserObject) )
            return null;
        
        TreeNodeUserObject nodeUserObject = (TreeNodeUserObject) userObject;
        for( int i=0; i<types.length; i++) {
            if ( types[i].getName().equals(nodeUserObject.getType()))
                return types[i].getIcon();
        }
        
        return null;
    }
    
    public Component getTreeCellRendererComponent(JTree tree, Object value,
        boolean sel,
        boolean expanded,
        boolean leaf, int row,
        boolean hasFocus) {
        Component ret = super.getTreeCellRendererComponent(tree, value, sel, expanded, leaf, row, hasFocus);
        
        DefaultMutableTreeNode treeNode = (DefaultMutableTreeNode) value;
        Object userObject = treeNode.getUserObject();
        if ( userObject == null) {
            setIcon(null);
            return ret;
        }
        
        setIcon(getIcon(userObject));
        
        setText(userObject.toString());
        
        return ret;
    }
    
}
